/*
 *   Copyright 2021 zzh
 *
 *   Licensed under the Apache License, Version 2.0 (the "License");
 *   you may not use this file except in compliance with the License.
 *   You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 *   Unless required by applicable law or agreed to in writing, software
 *   distributed under the License is distributed on an "AS IS" BASIS,
 *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *   See the License for the specific language governing permissions and
 *   limitations under the License.
 *
 */

package com.aduib.boot.common.concurrent;

import java.util.Map;
import java.util.Objects;
import java.util.concurrent.ConcurrentHashMap;

/**
 * @author ZZH
 * @version 1.0.0
 */
public class NameThreadLocalMap<T> {

  private Map<String, NameThreadLocal<T>> map;

  public NameThreadLocalMap() {
    this.map = new ConcurrentHashMap<>();
  }

  public NameThreadLocalMap(Map<String, NameThreadLocal<T>> map) {
    this.map = map;
  }

  public void addThreadLocal(NameThreadLocal<T> local) {
    NameThreadLocal<T> nameThreadLocal = map.get(local.getName());
    if (Objects.isNull(nameThreadLocal)) {
      map.put(local.getName(), local);
    } else {
      throw new RuntimeException(local.getName() + " threadLocal ready exists");
    }
  }

  public void removeThreadLocal(NameThreadLocal<T> local) {
    NameThreadLocal<T> nameThreadLocal = map.get(local.getName());
    if (Objects.nonNull(nameThreadLocal)) {
      nameThreadLocal.getThreadLocal().remove();
      nameThreadLocal = null;
      map.remove(local.getName(), local);
    }
  }
}
